bzoj2435 道路修建

最新推荐文章于 2018-02-20 15:39:00 发布

scarlyw

最新推荐文章于 2018-02-20 15:39:00 发布

阅读量338

点赞数

分类专栏： NOIP解题报告树的dfs bzoj刷题记录文章标签： bzoj NOI 暴力 dfs

本文链接：https://blog.csdn.net/scar_lyw/article/details/78334550

版权

NOIP解题报告同时被 3 个专栏收录

195 篇文章 0 订阅

订阅专栏

bzoj刷题记录

45 篇文章 1 订阅

订阅专栏

树的dfs

2 篇文章 0 订阅

订阅专栏

道路修建

题目背景：

bzoj2435

分析：暴力dfs

突如其来的傻题get·····直接dfs，每一个点对答案的贡献是，abs((n - size[cur])- size[cur]) * w（w为这个点与父亲的连边的权值），然后就没有然后了······

Source：

/*
	created by scarlyw
*/
#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <cctype>
#include <set>
#include <map>
#include <vector>
#include <queue>
#include <ctime>

const int MAXN = 1000000 + 10;

long long ans = 0;
int n, x, y, z;
int size[MAXN];

struct node {
	int to, w;
	node(int to = 0, int w = 0) : to(to), w(w) {}
} ;

std::vector<node> edge[MAXN];

inline void add_edge(int x, int y, int z) {
	edge[x].push_back(node(y, z)), edge[y].push_back(node(x, z));	
}

inline void read_in() {
	scanf("%d", &n);
	for (int i = 1; i < n; ++i) scanf("%d%d%d", &x, &y, &z), add_edge(x, y, z);
}

inline void dfs(int cur, int fa, int w) {
	size[cur] = 1;
	for (int p = 0; p < edge[cur].size(); ++p) {
		node *e = &edge[cur][p];
		if (e->to != fa) dfs(e->to, cur, e->w), size[cur] += size[e->to];
	}
	ans += (long long)abs(n - size[cur] - size[cur]) * (long long)w;
}

int main() {
	read_in();
	dfs(1, 0, 0);
	printf("%lld", ans);
	return 0;
}